#include <QCoreApplication>

#include "Network/DenseGradeWrapper.h"

int main(int argc, char *argv[])
{
    QString model_path = "../tensorrt-gpu-win-msvc2019/Script/export_dense121_gpu.engine";
    QString image_path = "../tensorrt-gpu-win-msvc2019/Script/1.jpg";

    DenseGradeWrapper denseWrapper(model_path);
    DenseInferResult res = denseWrapper.infer(image_path);

    qDebug() << "Level is: " << res.index;
    qDebug() << "Each possible is: ";
    for(int i = 0; i < 5; i++){
        qDebug() << "\t" << res.pie[i];
    }
}
